package com.sias.oj.manager;

import com.qcloud.cos.COSClient;
import com.qcloud.cos.model.PutObjectRequest;
import com.qcloud.cos.model.PutObjectResult;
import com.sias.oj.config.CosClientConfig;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.io.File;

/**
 * Cos 对象存储操作
 *
 * @author <a href="https://github.com/sogra">程序员sogra</a>
 */
@Component
public class CosManager {

  @Resource
  private CosClientConfig cosClientConfig;

  @Resource
  private COSClient cosClient;

  /**
   * 上传对象
   *
   * @param key           唯一键
   * @param localFilePath 本地文件路径
   * @return
   */
  public PutObjectResult putObject(String key, String localFilePath) {
    PutObjectRequest putObjectRequest = new PutObjectRequest(cosClientConfig.getBucket(), key,
            new File(localFilePath));
    return cosClient.putObject(putObjectRequest);
  }

  /**
   * 上传对象
   *
   * @param key  唯一键
   * @param file 文件
   * @return
   */
  public PutObjectResult putObject(String key, File file) {
    PutObjectRequest putObjectRequest = new PutObjectRequest(cosClientConfig.getBucket(), key,
            file);
    return cosClient.putObject(putObjectRequest);
  }
}
